#!/usr/bin/env python3
"""Alzheimer Latin America Analysis Script
Author: Juan Moises de la Serna
"""
import pandas as pd
import matplotlib.pyplot as plt

df = pd.read_csv('projections_2050.csv')

countries = ['Brazil','Mexico','Argentina','Colombia']
years = [2020,2030,2040,2050]

for _, row in df[df['country'].isin(countries)].iterrows():
    vals = [row[str(y)] for y in years]
    plt.plot(years, vals, marker='o', label=row['country'])

plt.title('Alzheimer Cases Projection to 2050')
plt.xlabel('Year')
plt.ylabel('Estimated Cases')
plt.legend()
plt.tight_layout()
plt.savefig('alzheimer_projection.png', dpi=150)
